# ################################################################
#                             HyperParameters
# ################################################################

class HyperParameters:
    # ################################################################
    #                             Data
    # ################################################################
    device = 'cpu'  # cpu
    data_root = 'D:/deepLearning_data/celeba'

    image_size = 64
    seed = 1234

    # ################################################################
    #                             Model
    # ################################################################
    z_dim = 100         # latency z dimension
    data_channels = 3   # RGB face

    # ################################################################
    #                             Exp
    # ################################################################
    batch_size = 144
    n_workers = 4       # data loader works : CPU kernels
    beta = 0.5          # adam optimizer 0.5 : GAN网络对超参数敏感，一般设计为0.5
    init_lr = 0.0002    # 初始学习率
    epochs = 1000       # 轮次数设计大一些
    verbose_step = 250  # evaluation: store image during training
    save_step = 1000    # save model step


HP = HyperParameters()
